﻿using OpenCvSharp;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace YoloPredictor
{
    public interface IPredictor
    {
        public List<Prediction> Predict(Mat picture, bool enableNMS = true);
    }
    public struct Prediction
    {
        /// <summary>
        /// 中心点坐标和大小
        /// </summary>
        public Rect2d Box;
        public int ClassId;
        public string ClassName;
        public float Probability, Confidence;
    }

    public struct PredictorConfig
    {
        public string Config, Names, Weights;
        public bool UseCUDA;
        public float Threshold, NmsThreshold;
    }
}
